from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    REDIS_HOST: str = "localhost"
    REDIS_PORT: int = 6379
    REDIS_DB: int = 0
    MODELS_DIR: str = "./models"  # 模型存储路径
    TEMP_FILES_DIR: str = "./temp"  # 临时文件路径
    GEMINI_API_KEY: str = ""  # 从.env加载AI密钥
    OLLAMA_URL: str = "http://118.145.200.164:11434"  # Ollama 的 API 地址
    OLLAMA_MODEL: str = "gemma3:4b"  # Ollama 的模型名称
    LMSTUDIO_URL: str = "http://localhost:1234"  # LM Studio 的 API 地址
    LMSTUDIO_MODEL: str = "google/gemma-3-4b"  # LM Studio 的模型名称
    AI_PROVIDER: str = "lmstudio"  # AI 提供者选择: "ollama" 或 "lmstudio"
    REDIS_URL: str = "redis://localhost:6379/0"  # 新添加这一行，处理额外输入

    class Config:
        env_file = ".env"
        extra = "allow"  # 可选：允许额外字段，避免类似问题；但优先添加字段


settings = Settings()

